package models

type File struct {
	ID       int    	`json:"id"`
	FileName string `json:"file_name"`
	MarkName string `json:"mark_name"`
	IsMarked int `json:"is_marked"`
}

func FileInsert(f File) error {
	err := db.Create(&f).Error

	if err != nil {
		return err
	}

	return nil
}

func FileMark(f File) error {
	err := db.Model(&f).Where("file_name = ?", f.FileName).Omit("id").UpdateColumns(f).Error
	if err != nil {
		return err
	}
	return nil
}

func FileExisted(fileName string) bool {
	var f File
	err := db.Select("id").Where("file_name = ?", fileName).Find(&f).Error
	if err != nil {
		return false
	}

	return true
}